/*
 * Copyright (c) 2021 Huawei Device Co., Ltd.
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

#ifndef MOCK_OHOS_ABILITY_RUNTIME_ABILITY_SCHEDULER_H
#define MOCK_OHOS_ABILITY_RUNTIME_ABILITY_SCHEDULER_H

#include "ability_scheduler_stub.h"
#include "ability_record.h"

namespace OHOS {
namespace AAFwk {
/**
 * @class AbilityScheduler
 * AbilityScheduler is used to schedule ability kit lifecycle.
 */
class AbilityScheduler : public AbilitySchedulerStub, virtual RefBase {
public:
    AbilityScheduler();
    virtual ~AbilityScheduler();

    void ScheduleAbilityTransaction(const Want& want, const LifeCycleStateInfo& targetState,
        sptr<SessionInfo> sessionInfo = nullptr) override;

    void ScheduleShareData(const int32_t &uniqueId) override;

    void SendResult(int requestCode, int resultCode, const Want& resultWant) override;

    const AbilityResult& GetResult() const;

    void ScheduleConnectAbility(const Want& want) override;

    void ScheduleDisconnectAbility(const Want& want) override;

    bool SchedulePrepareTerminateAbility() override;

    void ScheduleCommandAbility(const Want& want, bool restart, int startId) override;

    void ScheduleCommandAbilityWindow(const Want &want, const sptr<SessionInfo> &sessionInfo,
        WindowCommand winCmd) override;

    void ScheduleSaveAbilityState() override;

    void ScheduleRestoreAbilityState(const PacMap& inState) override;

    std::vector<std::string> GetFileTypes(const Uri& uri, const std::string& mimeTypeFilter) override;

    int OpenFile(const Uri& uri, const std::string& mode) override;

    int Insert(const Uri& uri, const NativeRdb::ValuesBucket& value) override;

    int Update(const Uri& uri, const NativeRdb::ValuesBucket& value,
        const NativeRdb::DataAbilityPredicates& predicates) override;

    int Delete(const Uri& uri, const NativeRdb::DataAbilityPredicates& predicates) override;

    std::shared_ptr<NativeRdb::AbsSharedResultSet> Query(
        const Uri& uri, std::vector<std::string>& columns, const NativeRdb::DataAbilityPredicates& predicates) override;

    std::shared_ptr<AppExecFwk::PacMap> Call(
        const Uri& uri, const std::string& method, const std::string& arg, const AppExecFwk::PacMap& pacMap) override;

    std::string GetType(const Uri& uri) override;

    int OpenRawFile(const Uri& uri, const std::string& mode) override;

    bool Reload(const Uri& uri, const PacMap& extras) override;

    int BatchInsert(const Uri& uri, const std::vector<NativeRdb::ValuesBucket>& values) override;

    Uri NormalizeUri(const Uri& uri) override;
    Uri DenormalizeUri(const Uri& uri) override;
    virtual bool ScheduleRegisterObserver(const Uri& uri, const sptr<IDataAbilityObserver>& dataObserver) override
    {
        return true;
    };
    virtual bool ScheduleUnregisterObserver(const Uri& uri, const sptr<IDataAbilityObserver>& dataObserver) override
    {
        return true;
    };
    virtual bool ScheduleNotifyChange(const Uri& uri) override
    {
        return true;
    };
    virtual std::vector<std::shared_ptr<AppExecFwk::DataAbilityResult>> ExecuteBatch(
        const std::vector<std::shared_ptr<AppExecFwk::DataAbilityOperation>>& operations) override
    {
        return std::vector<std::shared_ptr<AppExecFwk::DataAbilityResult>>();
    };
    virtual void NotifyContinuationResult(int32_t result) override
    {};
    virtual void ContinueAbility(const std::string& deviceId, uint32_t versionCode) override
    {};
    virtual void DumpAbilityInfo(const std::vector<std::string>& params, std::vector<std::string>& info) override
    {};
    virtual void CallRequest() override
    {
        return;
    };
    virtual void OnExecuteIntent(const Want &want) override
    {};
    virtual int CreateModalUIExtension(const Want &want) override
    {
        return 0;
    };
    virtual void UpdateSessionToken(sptr<IRemoteObject> sessionToken) override {}
private:
    AbilityResult result_;
};
}  // namespace AAFwk
}  // namespace OHOS
#endif  // MOCK_OHOS_ABILITY_RUNTIME_ABILITY_SCHEDULER_H
